python equivalent of javascript map